
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <termio.h>
#include <time.h>

/**
 *  uart模块插入虚拟机，生成ttyusb节点，应用程序往uart写入数据控制led
 *
 */



static int baudflag_arr[] = {
    B921600, B460800, B230400, B115200, B57600, B38400,
    B19200,  B9600,   B4800,   B2400,   B1800,  B1200,
    B600,    B300,    B150,    B110,    B75,    B50
};
static int speed_arr[] = {
    921600,  460800,  230400,  115200,  57600,  38400,
    19200,   9600,    4800,    2400,    1800,   1200,
    600,     300,     150,     110,     75,     50
};

int speed_to_flag(int speed)
{
    int i;
 
    for (i = 0;  i < sizeof(speed_arr)/sizeof(int);  i++) {
        if (speed == speed_arr[i]) {
            return baudflag_arr[i];
        }
    }
 
    fprintf(stderr, "Unsupported baudrate, use 9600 instead!\n");
    return B9600;
}

static struct termio oterm_attr;


//串口设置
int setup_port(int fd, int baud, int databits, int parity, int stopbits)
{
    struct termio term_attr;
 
    
    if (ioctl(fd, TCGETA, &term_attr) < 0) {
        return -1;
    }
 
    
    memcpy(&oterm_attr, &term_attr, sizeof(struct termio));
 
    term_attr.c_iflag &= ~(INLCR | IGNCR | ICRNL | ISTRIP);
    term_attr.c_oflag &= ~(OPOST | ONLCR | OCRNL);
    term_attr.c_lflag &= ~(ISIG | ECHO | ICANON | NOFLSH);
    term_attr.c_cflag &= ~CBAUD;
    term_attr.c_cflag |= CREAD | speed_to_flag(baud);
 
    
    term_attr.c_cflag &= ~(CSIZE);
    switch (databits) {
        case 5:
            term_attr.c_cflag |= CS5;
            break;
 
        case 6:
            term_attr.c_cflag |= CS6;
            break;
 
        case 7:
            term_attr.c_cflag |= CS7;
            break;
 
        case 8:
        default:
            term_attr.c_cflag |= CS8;
            break;
    }
 
    
    switch (parity) {
        case 1:  
            term_attr.c_cflag |= (PARENB | PARODD);
            break;
 
        case 2:  
            term_attr.c_cflag |= PARENB;
            term_attr.c_cflag &= ~(PARODD);
            break;
 
        case 0:  
        default:
            term_attr.c_cflag &= ~(PARENB);
            break;
    }
 
 
    
    switch (stopbits) {
        case 2:  
            term_attr.c_cflag |= CSTOPB;
            break;
 
        case 1:  
        default:
            term_attr.c_cflag &= ~CSTOPB;
            break;
    }
 
    term_attr.c_cc[VMIN] = 1;
    term_attr.c_cc[VTIME] = 0;
 
    if (ioctl(fd, TCSETAW, &term_attr) < 0) {
        return -1;
    }
 
    if (ioctl(fd, TCFLSH, 2) < 0) {
        return -1;
    }
 
    return 0;
}
 
 
void print_usage(char *program_name)
{
    fprintf(stderr,
            "*************************************\n"
            "  A Simple Serial Port Test Utility\n"
            "*************************************\n\n"
            "Usage:\n  %s <device> <baud> <databits> <parity> <stopbits> \n"
            "       databits: 5, 6, 7, 8\n"
            "       parity: 0(None), 1(Odd), 2(Even)\n"
            "       stopbits: 1, 2\n"
            "Example:\n  %s /dev/ttyUSB0 115200 8 0 1 64 00 3c\n\n",
            program_name, program_name
           );
}



unsigned char data_arr[50];
void byte_to_arr(char *vbuf)
{
	sscanf(vbuf, "%x %x %x %x %x %x %x %x",
	(unsigned int *)&data_arr[0], (unsigned int *)&data_arr[1], (unsigned int *)&data_arr[2], (unsigned int *)&data_arr[3],
	(unsigned int *)&data_arr[4], (unsigned int *)&data_arr[5], (unsigned int *)&data_arr[6], (unsigned int *)&data_arr[7]);
}



int main(int argc, char *argv[]) //./a.out /dev/ttyS4 115200 8 0 1 64 00 1c
{
    int fd;
    int baud;
    int count = 0;
    int databits;
    int stopbits;
    int parity;
	unsigned char red, green, blue;

    if (argc != 9) {
        print_usage(argv[0]);
        return 1;
    }
 
    baud = atoi(argv[2]);//获取波特率
    if ((baud < 0) || (baud > 921600)) {
        fprintf(stderr, "Invalid baudrate!\n");
        return 1;
    }
 
    databits = atoi(argv[3]); //获取数据位
    if ((databits < 5) || (databits > 8)) {
        fprintf(stderr, "Invalid databits!\n");
        return 1;
    }
 
    parity = atoi(argv[4]); //获取奇偶校验位
    if ((parity < 0) || (parity > 2)) {
        fprintf(stderr, "Invalid parity!\n");
        return 1;
    }
 
    stopbits = atoi(argv[5]); //获取停止位
    if ((stopbits < 1) || (stopbits > 2)) {
        fprintf(stderr, "Invalid stopbits!\n");
        return 1;
    }
 
 
    fd = open(argv[1], O_RDWR, 0); //打开串口设备文件
    if (fd < 0) {
        fprintf(stderr, "open <%s> error %s\n", argv[1], strerror(errno));
        return 1;
    }
 
    if (setup_port(fd, baud, databits, parity, stopbits)) { //设置串口
        fprintf(stderr, "setup_port error %s\n", strerror(errno));
        close(fd);
        return 1;
    }
	

	byte_to_arr("01 FE 01 02 F6 59 00 00"); //11bytes
	//byte_to_arr("01 FE 01 02 F6 59 00 00 64 00 3C"); //11bytes
	//byte_to_arr("01 FE 01 02 F6 59 00 00 00 00 00"); //11bytes
	sscanf(argv[6], "%x", (unsigned int *)&red);
	sscanf(argv[7], "%x", (unsigned int *)&green);
	sscanf(argv[8], "%x", (unsigned int *)&blue);
	printf("%s %s %s\n", argv[6], argv[7], argv[8]);
	//red = 0x64;
	//green = 0x0;
	//blue = 0x2c;
	
	//printf("%02x  %02x %02x \n", red, green, blue);
	while (1) {
		#if 1
		write(fd, &data_arr[count], 1);
		//printf("%02x over\n", data_arr[count]);
		count++;
		if (count > 7) {
			write(fd, &red, 	1);
			//printf("%02x over\n", red);
			count++;
			write(fd, &green, 	1);
			//printf("%02x over\n", green);
			count++;
			write(fd, &blue, 	1);
			//printf("%02x over\n", blue);
			count++;
		}
		if (count == 11) break;	
		
		#endif
		
	}
    close(fd);
 
    return 0;
}

